/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */

package cegg.graph.restriction;

import cegg.graph.vertex.ReasonVertex;
import java.awt.Graphics;
import java.io.Serializable;
import java.util.HashSet;
import java.util.Set;

/**
 *
 * @author kumper
 */
public abstract class Restriction implements Serializable {

    private Set<ReasonVertex> vertexSet;

    public Restriction(final ReasonVertex vertex1, final ReasonVertex vertex2)
    {
        this.vertexSet = new HashSet<ReasonVertex>();
        vertexSet.add(vertex1);
        vertexSet.add(vertex2);
    }

    public Restriction(final Set<ReasonVertex> set)
    {
        this.vertexSet = set;
    }

    public Set<ReasonVertex> getVertexSet() {
        return vertexSet;
    }

    public abstract void paint(Graphics g);

}
